package com.cic.wechat.server;

import com.cic.wechat.model.FileTypeHader;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.*;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @ProjectName:mywork
 * @Description: 加密文件转码
 * @Author: lingxiao 七面鬼
 * @Date: 2021/4/7 星期三 16:09
 * @Email: sheninglong@163.com
 */
@Service
public class Weigoto {

    @Value("${file.path.input}")
    String inpath;

    @Value("${file.path.out}")
    String outpath;


    /**
     * @ProjectName: com.cic.wechat.server
     * @Description: 二进制加密文件转换
     * @Author: lingxiao 七面鬼
     * @Date: 2021/4/13 16:40
     * @Email: sheninglong@163.com
     * @param path 文件路径
     * @param targetPath 文件输出路径
     * @return: null
     **/
    public void convert(){
        convert(inpath,outpath);
    }
    public void convert(String path, String targetPath) {

        File[] file = new File(path).listFiles();

        if (file == null) {

            return;

        }
        //文件偏移量信息
        AtomicInteger x = new AtomicInteger();

//        for (File file1 : file) {
//
//            if (file1.isFile()) {
//                //获取文件偏移量与文件类型
//                Object[] xori = getXor(file1);
//
//                if (xori != null && xori[1] != null) {
//
//                    x.set((int) xori[1]);
//
//                }
//
//                break;
//
//            }
//
//        }
        String ou=targetPath;

        Arrays.stream(file).parallel().forEach(file1 -> {

            if (file1.isDirectory()) {

                String[] newTargetPath = file1.getPath().split("/|\\\\");

                File targetFile = new File(ou + File.separator + newTargetPath[newTargetPath.length - 1]);

                if (!targetFile.exists()) {
                    System.out.println("创建" + targetFile.mkdirs());

                }
                convert(file1.getPath(), ou + File.separator + newTargetPath[newTargetPath.length - 1]);

            }

            Object[] xor = getXor(file1);

            if (x.get() == 0 && xor[1] != null && (int) xor[1] != 0) {

                x.set((int) xor[1]);

            }

            xor[1] = xor[1] == null ? x.get() : xor[1];

            toFile(file1,ou,xor);
        });
    }

    private void toFile(File file, String targetPath, Object[] xor){


        try (InputStream reader = new FileInputStream(file)) {
            Files.createDirectories(Paths.get(targetPath));
            OutputStream writer =
                    new FileOutputStream(targetPath + File.separator + file.getName().split("\\.")[0] + (xor[0] != null ?
                            "." + xor[0] : ""));
            byte[] bytes = new byte[1024 * 10];

            int b;
            //这里的in.read(bytes);就是把输入流中的东西，写入到内存中（bytes）。
            while ((b = reader.read(bytes)) != -1) {

                for (int i = 0; i < bytes.length; i++) {

                    bytes[i] = (byte) (int) (bytes[i] ^ (int) xor[1]);

                    if (i == (b - 1)) {

                        break;

                    }

                }
                writer.write(bytes, 0, b);
                writer.flush();
            }

            //integer.set(integer.get() + 1);

            //txtArea_log.append(file1.getName() + "(大小:" + ((double) file1.length() / 1000) + "kb,异或值:" + xor[1] + ")," + "进度：" + integer.get() + "/" + size + "\r\n");

        } catch (Exception e) {

            e.printStackTrace();

        }
    }

    /**
     * @ProjectName: com.cic.wechat.server
     * @Description: 获取文件偏移量和类型
     * @Author: lingxiao 七面鬼
     * @Date: 2021/4/13 16:37
     * @Email: sheninglong@163.com
     * @param file 加密文件
     * @return: null
     **/
    private static Object[] getXor(File file) {

        Object[] xor = null;

        if (file != null) {

            byte[] bytes = new byte[4];

            //读取文件文件头（前3位字节）
            try (InputStream reader = new FileInputStream(file)) {

                reader.read(bytes, 0, bytes.length);

            } catch (Exception e) {

                e.printStackTrace();

            }
            //获取偏移量和文件类型
            xor = getXor(bytes);

        }

        return xor;

    }


    /**
     * @ProjectName: com.cic.wechat.server
     * @Description: 计算16进制字节码数组的偏移量
     * @Author: lingxiao 七面鬼
     * @Date: 2021/4/13 16:45
     * @Email: sheninglong@163.com
     * @param bytes 需要计算偏移量的 16 进制字节码数组
     * @return: null
     **/
    private static Object[] getXor(byte[] bytes) {

        //存储文件头和类型
        Object[] xorType = new Object[2];
        //文件对应的文件头异或值
        int[] xors = new int[3];

        for (FileTypeHader value : FileTypeHader.values()) {

            String[] hex = {

                    String.valueOf(value.name().charAt(10)) + value.name().charAt(11),

                    String.valueOf(value.name().charAt(12)) + value.name().charAt(13),

                    String.valueOf(value.name().charAt(14)) + value.name().charAt(15)

            };

            xors[0] = bytes[0] & 0xFF ^ Integer.parseInt(hex[0], 16);

            xors[1] = bytes[1] & 0xFF ^ Integer.parseInt(hex[1], 16);

            xors[2] = bytes[2] & 0xFF ^ Integer.parseInt(hex[2], 16);

            //若偏移量相同则证明文件类型正确
            if (xors[0] == xors[1] && xors[1] == xors[2]) {
                //加密前的文件类型
                xorType[0] = value.getType();

                //加密的偏移量
                xorType[1] = xors[0];

                break;

            }

        }

        return xorType;

    }
}